'''
1 执行如下命令，安装所需的模块：
pip install numpy pandas matplotlib -i https://pypi.tuna.tsinghua.edu.cn/simple

2 创建两个文件：2015-2020.txt和2020.txt,内容分别从实验23的说明中复制

'''

import pandas as pd
import numpy as np
import matplotlib
import matplotlib.pyplot as plt

print("check if the backend name is Tkagg:",matplotlib.get_backend())
plt.rcParams['font.sans-serif'] = ['SimHei']

datas = pd.read_csv(r'2015-2020.txt', index_col=0)
datas = pd.DataFrame(datas.values, index=range(1, 11), columns=datas.columns)

plt.subplots(figsize=(16, 8))

plt.subplot(131)
for date in datas.columns:
    plt.plot(range(datas.shape[0]), datas[date], label=date)
#plt.ylim(0, 600000)
plt.legend()
plt.title('2015-2020年度票房Top10折线图\n作者：张三')

plt.subplot(132)
labels = range(2015, 2021)
x = np.arange(len(datas.columns))
i = 1
for index in datas.index:
    plt.bar(x + 0.1*i,datas.loc[index],width=0.1,tick_label = labels)
    i = i + 1
plt.title('2015-2020年度票房Top10柱状图')


plt.subplot(233)
plt.pie(datas['2019'], autopct='%1.1f%%')
plt.title('2019年度票房Top10饼图')

plt.subplot(236)
data2020 = pd.read_csv(r'2020.txt')
plt.pie(data2020['boxoffice'], autopct='%1.1f%%',labels=data2020['name'])
plt.title('2020年度票房Top10饼图')

plt.show()
